/**
* 版权所有:版权所有(C) 2020，卓信科技
* 系统名称:IFC-管理系统
* 文件名称:ZipFile.java
* 模块名称:文件压缩
* 创建日期:2020年5月19日
* 作        者:Baisong
* 内容摘要:文件压缩
*/
package com.zx.emanage.util.gen;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import javax.servlet.http.HttpServletResponse;

import com.zx.emanage.util.gen.entity.BatchCompressImgVO;
import com.zx.sframe.util.file.GlobalFileUtil;

/**
 * 类  名  称:ZipFile
 *
 * 内容摘要:TODO(文件压缩)
 *
 * 作        者:Baisong
 */
public class ZipFile
{
	
	private static List<BatchCompressImgVO> fileList = new ArrayList<>();
	
	/***
	 * // 文件打包下载
	 */
	public static HttpServletResponse downLoadFiles(List<File> files,
			HttpServletResponse response,List<BatchCompressImgVO> attList) throws Exception 
	{
		try 
		{
			fileList = attList;
			/**
			 * 创建一个临时压缩文件， 我们会把文件流全部注入到这个文件中 这里的文件你可以自定义是.rar还是.zip
			 */
			File file = File.createTempFile("ImgsZip", ".zip");
			//判断
			if (!file.exists())
{
				file.createNewFile();
			}
			response.reset();
			// response.getWriter()
			// 创建文件输出流
			FileOutputStream fous = new FileOutputStream(file);
			/**
			 * 打包的方法我们会用到ZipOutputStream这样一个输出流, 所以这里我们把输出流转换一下
			 */
			ZipOutputStream zipOut = new ZipOutputStream(fous);
			/**
			 * 这个方法接受的就是一个所要打包文件的集合， 还有一个ZipOutputStream
			 */
			zipFile(files, zipOut);
			zipOut.close();
			fous.close();
			return downloadZip(file, response);
		}
		catch (Exception e) 
		{
			e.printStackTrace();
		}
		/**
		 * 直到文件的打包已经成功了， 文件的打包过程被我封装在FileUtil.zipFile这个静态方法中，
		 * 稍后会呈现出来，接下来的就是往客户端写数据了
		 */

		return response;
	}

	/**
	 * 把接受的全部文件打成压缩包
	 * 
	 * @param List
	 *            <File>;
	 * @param org
	 *            .apache.tools.zip.ZipOutputStream
	 */
	public static void zipFile(List<File> files, ZipOutputStream outputStream)
	{
		int size = files.size();
		//循环
		for (int i = 0; i < size; i++) 
		{
			File file = (File) files.get(i);
			zipFile(file, outputStream);
		}
	}
	/**
	 * 文件下载
	 */
	public static HttpServletResponse downloadZip(File file,
			HttpServletResponse response) 
	{
		try
		{
			// 以流的形式下载文件。
			InputStream fis = new BufferedInputStream(new FileInputStream(
					file.getPath()));
			byte[] buffer = new byte[fis.available()];
			fis.read(buffer);
			fis.close();
			// 清空response
			response.reset();

			OutputStream toClient = new BufferedOutputStream(
					response.getOutputStream());
			response.setContentType("application/octet-stream");

			// 如果输出的是中文名的文件，在此处就要用URLEncoder.encode方法进行处理
			response.setHeader("Content-Disposition", "attachment;filename="
					+ URLEncoder.encode(file.getName(), "UTF-8"));
			toClient.write(buffer);
			toClient.flush();
			toClient.close();
		}
		catch (IOException ex)
		{
			ex.printStackTrace();
		}
		finally 
		{
			try
			{
				File f = new File(file.getPath());
				f.delete();
			} 
			catch (Exception e)
			{
				e.printStackTrace();
			}
		}
		return response;
	}

	/**
	 * 根据输入的文件与输出流对文件进行打包
	 * 
	 * @param File
	 * @param org
	 *            .apache.tools.zip.ZipOutputStream
	 */
	public static void zipFile(File inputFile, ZipOutputStream ouputStream)
	{
		try
		{
			//判断
			if (inputFile.exists())
			{
				/**
				 * 如果是目录的话这里是不采取操作的， 至于目录的打包正在研究中
				 */
				if (inputFile.isFile()) 
				{
					FileInputStream IN = new FileInputStream(inputFile);
					BufferedInputStream bins = new BufferedInputStream(IN, 512);
					// org.apache.tools.zip.ZipEntry
					ZipEntry entry = new ZipEntry(
							getNewName(inputFile.getName()));
					ouputStream.putNextEntry(entry);
					// 向压缩文件中输出数据
					int nNumber;
					byte[] buffer = new byte[512];
					//循环
					while ((nNumber = bins.read(buffer)) != -1) 
					{
						ouputStream.write(buffer, 0, nNumber);
					}
					// 关闭创建的流对象
	                    bins.close();
	                    IN.close();
	                } 
					else 
	                {
	                    try 
	                    {
	                        File[] files = inputFile.listFiles();
	                        //循环
	                        for (int i = 0; i < files.length; i++) 
	                        {
	                            zipFile(files[i], ouputStream);
	                        }
	                    } 
	                    catch (Exception e)
	                    {
	                        e.printStackTrace();
	                    }
	                }
	            }
	        }
			catch (Exception e)
			{
	            e.printStackTrace();
	        }
	    }

	/**
	 * 
	 * @Title: getNewName
	 * @Description: TODO(获取名字)
	 * @param addrss_name
	 * @return
	 * @author: baisong
	 * @time:2016年11月2日 下午3:29:23 history: 1、2016年11月2日 baisong 创建方法
	 */
	public static String getNewName(String addrss_name)
	{
		String name = addrss_name;
		String type = addrss_name;
		//循环
		for (BatchCompressImgVO vo : fileList)
		{
			//判断
			if (GlobalFileUtil.getFileName(vo.getAttachment_address()).equals(
					addrss_name)) 
			{
				//判断
				if(addrss_name.indexOf(".")>=0)
				{
					type = addrss_name.substring(addrss_name.lastIndexOf(".")+1);
				}
				name = vo.getAttachment_old_name() + "." + type;
			}
		}
		return replacePath(name);
	}

	// 去掉/ 避免创建文件夹
	private static String replacePath(String imgName)
	{
		return imgName.replace("/", "");
	}

}

